/*
 *  Copyright 2021 Sonu Kumar
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         https://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and limitations under the License.
 *
 */

package com.github.sonus21.rqueue.test.common;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.BeforeEach;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.test.web.reactive.server.HttpHandlerConnector;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.server.adapter.WebHttpHandlerBuilder;

public abstract class SpringWebTestBase extends SpringTestBase {

  @Autowired(required = false)
  protected WebApplicationContext wac;

  protected WebClient webClient;
  protected MockMvc mockMvc;
  @Autowired protected ObjectMapper mapper;
  @Autowired ApplicationContext applicationContext;

  @BeforeEach
  public void init() {
    if (reactiveEnabled) {
      webClient =
          WebClient.builder()
              .clientConnector(
                  new HttpHandlerConnector(
                      WebHttpHandlerBuilder.applicationContext(applicationContext).build()))
              .build();
    } else {
      this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
    }
  }
}
